# 设置LSTM训练参数
class TrainingConfig(object):
    train_batch_size = 5
    dev_batch_size = 20
    lr = 0.001  # 学习率
    epoch = 20


class LSTMConfig(object):
    embedding_num = 80  # 词向量维数
    hidden_num = 80  # 隐向量维数
